package com.lab.cloud.common.validation.util;

import org.apache.commons.lang3.StringUtils;

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.util.Arrays;

public class CNIDValidator implements ConstraintValidator<CNID, String> {

	@Override
	public void initialize(CNID constraintAnnotation) {

	}

	@Override
	public boolean isValid(String num, ConstraintValidatorContext context) {
		num = StringUtils.trimToEmpty(num).toUpperCase();

		if (StringUtils.isBlank(num)) {
			return false;
		}

		char[] id = {};
		for (int i = 0; i < num.length(); i++) {
			id = Arrays.copyOf(id, id.length + 1);
			id[id.length - 1] = num.charAt(i);
		}

		boolean IsFormRight = verForm(num);

		if (IsFormRight) {
			boolean IsCorrect = verify(id);
			if (IsCorrect) {
				return true;
			} else {
				return false;
			}
		} else {
			return false;
		}

	}

	/**
	 * 验证身份证格式
	 * 
	 * @param num
	 * @return
	 */
	public static boolean verForm(String num) {
		String reg = "^\\d{15}$|^\\d{17}[0-9Xx]$";
		if (!num.matches(reg)) {
			System.out.println("Format Error!");
			return false;
		}
		return true;
	}

	/**
	 * 验证身份证尾号
	 * 
	 * @param id
	 * @return
	 */
	public static boolean verify(char[] id) {
		int sum = 0;
		int w[] = { 7, 9, 10, 5, 8, 4, 2, 1, 6, 3, 7, 9, 10, 5, 8, 4, 2 };
		char[] ch = { '1', '0', 'X', '9', '8', '7', '6', '5', '4', '3', '2' };
		for (int i = 0; i < id.length - 1; i++) {
			sum += (id[i] - '0') * w[i];
		}
		int c = sum % 11;
		char code = ch[c];
		char last = id[id.length - 1];
		last = last == 'x' ? 'X' : last;
		return last == code;
	}

}